Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_schema_migration_upgrade.py

Displaying Raw • Download

T8b949e# SPDX-License-Identifier: 0BSD

T8b949e"""Upgrade path: simulate database at version N-1 and assert migration to LATEST_VERSION."""

Tff7b72import T7ee787pytest

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787provider Tff7b72import Te6edf3DatabaseProvider

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787schema Tff7b72import Te6edf3DatabaseSchema

Tff7b72def Td2a8ff_column_namesTb4b4b4(Te6edf3providerTb4b4b4, Te6edf3tableTb4b4b4: Tffa657strTb4b4b4) Tff7b72-Tff7b72> Tffa657setTb4b4b4[Tffa657strTb4b4b4]Tb4b4b4:

Te6edf3cur Tff7b72= Te6edf3providerTff7b72.Td2a8ffconnectionTff7b72.Td2a8ffcursorTb4b4b4(Tb4b4b4)

Tff7b72tryTb4b4b4:

Te6edf3curTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffPRAGMA table_info(Tffd700{Te6edf3tableTffd700}Ta5d6ff)Ta5d6ff"Tb4b4b4)

Tff7b72return Tb4b4b4{Te6edf3rowTb4b4b4[T79c0ff1Tb4b4b4] Tff7b72for Te6edf3row Tff7b72in Te6edf3curTff7b72.Td2a8fffetchallTb4b4b4(Tb4b4b4)Tb4b4b4}

Tff7b72finallyTb4b4b4:

Te6edf3curTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffskipifTb4b4b4(

Te6edf3DatabaseSchemaTff7b72.Td2a8ffLATEST_VERSION Tff7b72< T79c0ff41Tb4b4b4,

Te6edf3reasonTff7b72=Ta5d6ff"Ta5d6ffTest targets migration block for version 41Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72def Td2a8fftest_migrate_from_version_40_restores_attachments_strippedTb4b4b4(Te6edf3tmp_pathTb4b4b4)Tb4b4b4:

T8b949e"""Version 41 adds lxmf_messages.attachments_stripped; re-apply from 40."""

Te6edf3db_path Tff7b72= Te6edf3tmp_path Tff7b72/ Ta5d6ff"Ta5d6ffmig.dbTa5d6ff"

Te6edf3provider Tff7b72= Te6edf3DatabaseProviderTb4b4b4(Tffa657strTb4b4b4(Te6edf3db_pathTb4b4b4)Tb4b4b4)

Te6edf3schema Tff7b72= Te6edf3DatabaseSchemaTb4b4b4(Te6edf3providerTb4b4b4)

Te6edf3schemaTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

Tff7b72assert Tb4b4b4(

Tffa657intTb4b4b4(

Te6edf3providerTff7b72.Td2a8fffetchoneTb4b4b4(

Ta5d6ff"Ta5d6ffSELECT value FROM config WHERE key = ?Ta5d6ff"Tb4b4b4,

Tb4b4b4(Ta5d6ff"Ta5d6ffdatabase_versionTa5d6ff"Tb4b4b4,Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4[Ta5d6ff"Ta5d6ffvalueTa5d6ff"Tb4b4b4]Tb4b4b4,

Tb4b4b4)

Tff7b72== Te6edf3DatabaseSchemaTff7b72.Td2a8ffLATEST_VERSION

Tb4b4b4)

Te6edf3providerTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffUPDATE config SET value = ? WHERE key = ?Ta5d6ff"Tb4b4b4,

Tb4b4b4(Ta5d6ff"Ta5d6ff40Ta5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffdatabase_versionTa5d6ff"Tb4b4b4)Tb4b4b4,

Tb4b4b4)

Tff7b72tryTb4b4b4:

Te6edf3providerTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"Ta5d6ffALTER TABLE lxmf_messages DROP COLUMN attachments_strippedTa5d6ff"Tb4b4b4)

Tff7b72except Tf85149Exception Tff7b72as Te6edf3excTb4b4b4:

Te6edf3pytestTff7b72.Td2a8ffskipTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffSQLite DROP COLUMN not available or failed: Tffd700{Te6edf3excTffd700}Ta5d6ff"Tb4b4b4)

Tff7b72assert Ta5d6ff"Ta5d6ffattachments_strippedTa5d6ff" Tff7b72not Tff7b72in Te6edf3_column_namesTb4b4b4(Te6edf3providerTb4b4b4, Ta5d6ff"Ta5d6fflxmf_messagesTa5d6ff"Tb4b4b4)

Te6edf3schemaTff7b72.Td2a8ffmigrateTb4b4b4(Te6edf3schemaTff7b72.Td2a8ff_get_current_versionTb4b4b4(Tb4b4b4)Tb4b4b4)

Tff7b72assert Ta5d6ff"Ta5d6ffattachments_strippedTa5d6ff" Tff7b72in Te6edf3_column_namesTb4b4b4(Te6edf3providerTb4b4b4, Ta5d6ff"Ta5d6fflxmf_messagesTa5d6ff"Tb4b4b4)

Tff7b72assert Tb4b4b4(

Tffa657intTb4b4b4(

Te6edf3providerTff7b72.Td2a8fffetchoneTb4b4b4(

Ta5d6ff"Ta5d6ffSELECT value FROM config WHERE key = ?Ta5d6ff"Tb4b4b4,

Tb4b4b4(Ta5d6ff"Ta5d6ffdatabase_versionTa5d6ff"Tb4b4b4,Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4[Ta5d6ff"Ta5d6ffvalueTa5d6ff"Tb4b4b4]Tb4b4b4,

Tb4b4b4)

Tff7b72== Te6edf3DatabaseSchemaTff7b72.Td2a8ffLATEST_VERSION

Tb4b4b4)

Te6edf3providerTff7b72.Td2a8ffclose_allTb4b4b4(Tb4b4b4)